import time
from common.base import *

# 打开浏览器
driver = webdriver.Chrome()
# 请求百度
driver.get('https://www.baidu.com')
# 截图 保存在当前目录下
# driver.get_screenshot_as_file('文件路径+文件名')
# 动态获取文件名称 使用时间戳
driver.get_screenshot_as_file('./%s.png' % (time.strftime('%Y_%m-%d %H_%M_%S')))
# 退出浏览器
sleep(2)
driver.quit()